from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension


setup(
    name="lzs_cpp", 
    version="1.2", 
    description="CUDA kernel fusion exercise", 
    url="https://zgsb.com", 
    author="Xiaobao Li", 
    author_email="caibozongzi@163.com", 
    ext_modules=[
        CppExtension("lzs_cpp", ["my_test_1.cpp"])
    ],
    cmdclass={
        "build_ext": BuildExtension
    })
